import { NodeConfig } from './nodeConfig'
import { EdgeConfig } from './edgeConfig'

/**
 * 示例数据
 */
export class SampleData {
  /**
   * 创建示例节点
   * @param {Object} graphManager 图形管理器
   */
  static createSampleNodes(graphManager) {
    // 开始节点
    const startNode = graphManager.createNodeByType('start', '开始', 100, 200)
    
    // 结束节点
    const endNode = graphManager.createNodeByType('end', '结束', 400, 200)
    
    // 连接开始和结束节点
    try {
      const edgeConfig = EdgeConfig.getEdgeConfig(startNode, endNode, 'normal')
      const edge = graphManager.addEdge(edgeConfig)
      
      // 为示例连线添加加号图标
      setTimeout(() => {
        graphManager.addPlusIconToEdge(edge)
      }, 200)
    } catch (error) {
      console.warn('连接开始和结束节点失败:', error)
    }

    return { startNode, endNode }
  }
} 